package org.wheel.jpa.boot.internal;

import org.hibernate.jpa.spi.IdentifierGeneratorStrategyProvider;
import org.wheel.jpa.id.IntegerIdGenerator;
import org.wheel.jpa.id.LongIdGenerator;
import org.wheel.jpa.id.StringIdGenerator;

import java.util.HashMap;
import java.util.Map;

/**
 * @author klaus.wang
 */
public class WebIdentifierGeneratorStrategyProvider implements IdentifierGeneratorStrategyProvider {

	@Override
	public Map<String, Class<?>> getStrategies() {
		Map<String, Class<?>> strategies = new HashMap<>(3);
		strategies.put("snow.l", LongIdGenerator.class);
		strategies.put("snow.i", IntegerIdGenerator.class);
		strategies.put("snow.s", StringIdGenerator.class);
		return strategies;
	}

}
